function [good_vec average_vec] = get_good_avg_vec(Y, I, good, average);

good_vec = [];
average_vec = [];

for i = 1:length(Y)
    if(Y(i) < good)
        good_vec = [good_vec I(i)];
    else if(Y(i) < average)
            average_vec = [average_vec I(i)];
        end
    end
end